#include-once
#include <corz_essentials.au3>
global $handle, $zip_loaded, $zipper

; Zip.dll wrapper functions..
;
; These functions are designed to utilize the common capabilities of corz zip.dll
; (as distributed with corz apps installer)
;
; Applications using these functions get drop-in access to all the zipper capabilities.
;
; Cor @ corz.org 2008->tomorrow! ;o)
;


; MakeZip()
;
; Returns an array of two values. The first is the return code from the zipper
; the second is the contents of the @error macro.
;
; Normally, MakeZip() returns a text message describing the state. "ok", if everything
; went okay, or some error message if it didn't.
;
; You can also access the return codes directly, by passing the optional fourth parameter.
; This is handy if you want to intercept them for some reason - you can always use
; ProcessZipCodes() afterwards.
;
; Set the fourth parameter to true, and get back the raw values as an array. This array
; has two values, the first, the actual return code from the zipper, and the second,
; the contents of @error, if there are any.
;
;	$foo = MakeZip($src, $zip, "", 1)
;	$bar = $foo[0]
;	$foo = ProcessZipCodes($foo)
;
; See ProcessZipCodes() below, for a fuller description of the various codes that the
; module creates.
;
func MakeZip($source, $target, $pass="", $raw=0)

	local $result, $ret[2]

	if not $zip_loaded then $zip_loaded = RegisterZipDLL($zipper, $zip_loaded)
	if $zip_loaded < 1 then
		$ret[1] = -2
		return false
	endif


	if $pass then
		$new_zip = ZipCreate($target, $pass) ; $new_zip => handle, expressed as 0x001428**
	else
		$new_zip = ZipCreate($target)
	endif

	; couldn't create zip..
	if Number($new_zip) == 0 then ;	0x00000000
		$result = -8
		SetError(-8)
	else
		; okay..
		if IsDir($source) then
			if StringRight($source, 1) = "\" then $source = StringTrimRight($source, 1)
			$result = ZipAddDir($new_zip, $source, 1)
		else
			$inner = BaseName($source)
			$result = ZipAdd($new_zip, $source, $inner)
		endif
		switch Number($result)
			case 512
				$result = -6
			case 0
				$result = 1
		endswitch
		ZipClose($new_zip)	; = 0

	endif

	$ret[0] = $result
	$ret[1] = @error

	if $raw then return $ret
	return ProcessZipCodes($ret)

endfunc



; UnZip a file, defaults to current location
func UnZip($archive, $destination="", $pass="", $raw=0)

	local $result, $ret[2]

	if not $zip_loaded then $zip_loaded = RegisterZipDLL($zipper, $zip_loaded)
	if $zip_loaded < 1 then return false

	if not $destination then $destination = GetParent($archive)
	$result = ZipUnZip($archive, $destination, $pass)

	$ret[0] = $result
	$ret[1] = @error

	if $raw then return $ret
	return ProcessZipCodes($ret)

endfunc


; RegisterZipDLL()
; get the zipper dll ready for use..
;
; return values:
;	0	=	zipper could not be found
;	1	=	successful loaded corzip plugin [default]
;
func RegisterZipDLL($zipper, $loaded=0)

	if $loaded then return $loaded
	if not FileExists($zipper) then return 0

	$handle = PluginOpen($zipper)
	Sleep(200)
	if PluginClose($handle) then
		$handle = PluginOpen($zipper)
		return 1
	endif
	return -1
endfunc



func ProcessZipCodes($val_arr, $unz=false)

	if not IsArray($val_arr) then return "Not a valid ZipCodes array"

	local $code = $val_arr[0]
	local $err = $val_arr[1]

	local $msg
	if not $zip_loaded then
		$msg = "there was a problem loading the zipper"
	else
		switch $code
			case 0
				if $unz then
					$msg = "ok"
				else
					$msg = "some error"
				endif
			case -6
				$msg = "couldn't add a file to the zip"
			case -8
				$msg = "couldn't create the zip file"
			case else
				; normally 100663296 or something.	[perhaps problem casting in C++] ;2do
				$msg = "ok"
		endswitch
	endif
	return $msg
endfunc



;
; These are from the UDF that came with Stephen Podhajecki's original plugin..
;

;===============================================================================
; Function Name	:	_ZipList2Array
; Description		:	Retrieves and formats the Item Info for all items in the zip
;							into a 2 dimensional array.
; Parameter(s)		:	$szZipFile	The ZipFile to retrieve the item info from.
;
; Requirement(s)	:	Au3Zip.dll Autoit v3.2.2.0 or >
; Return Value(s)	:	2 dimensional array on success for empty string and @error =1
; User CallTip		:
; Author(s)			:	Stephen Podhajecki <gehossafats at netmdc.com/>
; Note(s)			:
;===============================================================================
Func _ZipList2Array($szZipFile)
	Local $zCount, $zipItems[1][8]
	If FileExists($szZipFile) Then
		$zCount = ZipGetCount($szZipFile)
		If $zCount Then
			ReDim $zipItems[$zCount][8]
			For $z_x = 0 To $zCount - 1
				Local $zipTemp = _ZipItemInfo2Array($szZipFile, $z_x)
				If Not (@error) Then
					For $z = 0 To 7
						$zipItems[$z_x][$z] = $zipTemp[$z]
					Next
				EndIf
			Next
		EndIf
		Return $zipItems
	EndIf
	Return SetError(1, 0, "")
EndFunc   ;==>_ZipList2Array


;===============================================================================
; Function Name	:	_ZipItemInfo2Array
; Description		:	Returns and array of zip item info.
; Parameter(s)		:	$szZipFile	The zip file to get the item info from
;							$index		The index of the item to get.
;
; Requirement(s)	:	Au3Zip.dll  Autoit v3.2.2.0
; Return Value(s)	:	An Single dim Array contain the item info, error 1 and empty on fail.
; User CallTip		:
; Author(s)			:	Stephen Podhajecki <gehossafats at netmdc.com/>
; Note(s)			:	This function is called by _ZipList2Array()
;===============================================================================
Func _ZipItemInfo2Array($szZipFile, $index)
	Local $z_v_ret, $strList, $zipTemp, $zipItems[8]
	If FileExists($szZipFile) Then
		If $index >= 0 Then
			If $index <= ZipGetCount ($szZipFile) Then
				$z_v_ret = ZipGetItemInfo ($szZipFile, $index)
				If $z_v_ret <> 0 Then
					Return _ZipGetInfoFromPtr($z_v_ret)
				EndIf
			EndIf
		EndIf
	EndIf
	Return SetError(1, 0, 0)
EndFunc   ;==>_ZipItemInfo2Array


;===============================================================================
; Function Name	:	_ZipGetInfoFromPtr
; Description		:
; Parameter(s)		:	$zPtr	Pointer to the zip entry struct.
;
; Requirement(s)	:	Au3Zip.dll AutoIt v3.2.20
; Return Value(s)	:	Array of zip item info, @error =1 and 0 on fail
; User CallTip		:
; Author(s)			:	Stephen Podhajecki <gehossafats at netmdc.com/>
; Note(s)			:	Called by other fuctions.
;===============================================================================
Func _ZipGetInfoFromPtr($zPtr)
;==============================================================================
;~ typedef struct
;~ { int index;                 // index of this file within the zip
;~   TCHAR name[MAX_PATH];      // filename within the zip
;~   DWORD attr;                // attributes, as in GetFileAttributes.
;~   FILETIME atime,ctime,mtime;// access, create, modify filetimes
;~   long comp_size;            // sizes of item, compressed and uncompressed. These
;~   long unc_size;             // may be -1 if not yet known (e.g. being streamed in)
;~ } ZIPENTRY;
;==============================================================================
	If $zPtr <> 0 Then
		Local $zipItems[8]
		$strList = DllStructCreate("int;char[260];dword;int64;int64;int64;long;long", $zPtr)
		For $z = 1 To 8
			Local $zipTemp = DllStructGetData($strList, $z)
			If $z > 3 And $z < 7 Then
				$zipItems[$z - 1] = _FileTime2SystemTimeZ($zipTemp)
			Else
				$zipItems[$z - 1] = $zipTemp
			EndIf
		Next
		$strList = 0
		Return $zipItems
	EndIf
EndFunc


;===============================================================================
; Function Name:	_FileTime2SystemTimeZ
; Description:		Converts and formats Filetime to Systemtime.
; Parameter(s):			$file_t		64 bit filetime returned from zip archive.
; Requirement(s):
; Return Value(s):	Hopefully a nice formatted string mm/dd/yyyy hr:mim:sec[A/P]m
; User CallTip:
; Author(s):		Stephen Podhajecki <gehossafats at netmdc.com/>
; Note(s):	99% sure this is correct.
;===============================================================================
Func _FileTime2SystemTimeZ($file_t)
	Local $td, $st, $ft, $vret, $syst, $zone, $hr
	$td = "%s/%s/%s %s:%s:%s%s" ;date and time format
	If $file_t = "" Then Return
	$st = DllStructCreate("short;short;short;short;short;short;short;short")
	$ft = DllStructCreate("dword;dword")
	DllStructSetData($ft, 1, _Lo_dWordZ($file_t))
	DllStructSetData($ft, 2, _Hi_dWordZ($file_t))
	$vret = DllCall("kernel32.dll", "int", "FileTimeToSystemTime", "ptr", DllStructGetPtr($ft), "ptr", DllStructGetPtr($st))
	If IsArray($vret) Then
		$zone = "AM"
		$hr = DllStructGetData($st, 5)
		If $hr > 11 Then $zone = "PM"
		If $hr > 12 Then $hr -= 12

		$syst = StringFormat($td, _ZipPZ(DllStructGetData($st, 2)), _
				_ZipPZ(DllStructGetData($st, 4)), _
				DllStructGetData($st, 1), _
				_ZipPZ($hr), _
				_ZipPZ(DllStructGetData($st, 6)), _
				_ZipPZ(DllStructGetData($st, 7)), _
				$zone)

		;ConsoleWrite($syst&@LF)
	EndIf
	$st = 0
	$ft = 0
	Return $syst
EndFunc   ;==>_FileTime2SystemTimeZ
;===============================================================================
; Function Name:	_Hi_dWordZ
; Description:	Get Hi order 32 bits from 64 bit number
; Parameter(s):			$file_t		 filetime
; Requirement(s):
; Return Value(s):	Return Hi order 32 bits from 64 bit number
; User CallTip:
; Author(s):		Stephen Podhajecki <gehossafats at netmdc.com/>
; Note(s):			used to convert filetime to systemtime.
;===============================================================================
Func _Hi_dWordZ($file_t)
	;ConsoleWrite("Hi :"&$file_t/(2^32)&@LF)
	Return $file_t/ (2 ^ 32)
EndFunc   ;==>_Hi_dWordZ
;===============================================================================
; Function Name:	_Lo_dWordZ
; Description:		Gets low order 32 bits of 64 bit number.
; Parameter(s):			$file_t		 Filetime
; Requirement(s):
; Return Value(s):	Lo order 32 bits of 64 bit number.
; User CallTip:
; Author(s):		Stephen Podhajecki <gehossafats at netmdc.com/>
; Note(s):
;===============================================================================
Func _Lo_dWordZ($file_t)
	;ConsoleWrite("Lo:"&abs($file_t - ((2^32) * ($file_t/(2^32))))&@LF)
	Return Abs($file_t - ((2 ^ 32) * ($file_t/ (2 ^ 32))))
EndFunc   ;==>_Lo_dWordZ
;===============================================================================
; Function Name:	_ZipPZ
; Description:		adds a"0" prefix to numbers >=0 and <10
; Parameter(s):			$Value
; Requirement(s):
; Return Value(s):	Modified value.
; User CallTip:
; Author(s):		Stephen Podhajecki <gehossafats at netmdc.com/>
; Note(s):
;===============================================================================
Func _ZipPZ($Value)
	Select
		Case $Value < 0
			Return "00"
		Case $Value < 10
			Return "0" & $Value
	EndSelect
	Return $Value
EndFunc   ;==>_ZipPZ